import streamlit as st


# 定义一个执行基本算术运算的函数，作为测试服务
def arithmetic_service(operation, a, b):
    if operation == 'add':
        return a + b
    elif operation == 'subtract':
        return a - b
    elif operation == 'multiply':
        return a * b
    elif operation == 'divide':
        if b != 0:
            return a / b
        else:
            return "Error: Division by zero"
    else:
        return "Error: Unknown operation"

    # 设置页面标题


st.title('Arithmetic Test Service')

# 创建一个文本输入框用于输入第一个数字
num1 = st.number_input('Enter first number', min_value=0, step=1)

# 创建一个文本输入框用于输入第二个数字
num2 = st.number_input('Enter second number', min_value=0, step=1)

# 创建一个选择框用于选择算术运算
operation = st.selectbox('Select operation', ['add', 'subtract', 'multiply', 'divide'])

# 创建一个按钮，点击时调用算术服务并显示结果
if st.button('Calculate'):
    result = arithmetic_service(operation, num1, num2)
    st.write('Result:', result)
